import sqlite3
import os

# 数据库文件路径
DATABASE = os.path.join(os.path.dirname(__file__), 'packages.db')

# 初始化数据库
def init_db():
    # 确保数据库文件不存在
    if os.path.exists(DATABASE):
        os.remove(DATABASE)
    
    # 连接数据库
    conn = sqlite3.connect(DATABASE)
    cursor = conn.cursor()
    
    # 读取schema.sql文件
    with open('schema.sql', 'r', encoding='utf-8') as f:
        schema = f.read()
    
    # 执行SQL语句
    cursor.executescript(schema)
    
    # 提交更改
    conn.commit()
    
    # 关闭连接
    conn.close()
    
    print("数据库初始化完成")

if __name__ == "__main__":
    init_db()
